#!/usr/bin/env python3
# -*- coding: utf-8 -*-

class BaseSchedulerPlugin:
    """学习率调度器插件基类"""
    
    def get_lr(self, current_step, total_steps, base_lr):
        """
        计算当前学习率
        
        Args:
            current_step (int): 当前步数
            total_steps (int): 总步数
            base_lr (float): 基础学习率
            
        Returns:
            float: 当前学习率
        """
        raise NotImplementedError("子类必须实现 get_lr 方法")